#pragma once
#include <torch/torch.h>
#include <string>

namespace model {
    torch::jit::Module load_model(const std::string& model_path);
} 